package com.itcast.bos.web.action;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;

import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.struts2.ServletActionContext;
import org.apache.struts2.convention.annotation.Action;
import org.apache.struts2.convention.annotation.Namespace;
import org.apache.struts2.convention.annotation.ParentPackage;
import org.apache.struts2.convention.annotation.Result;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Controller;

import com.itcast.bos.domain.Area;
import com.itcast.bos.service.AreaService;
import com.itcast.bos.utils.PinYin4jUtils;
import com.itcast.bos.web.action.base.BaseAction;
import com.opensymphony.xwork2.ActionContext;
import com.opensymphony.xwork2.ActionSupport;
import com.opensymphony.xwork2.ModelDriven;

@Controller
@Scope("prototype")
@Namespace("/")
@ParentPackage("json-default")
public class AreaAction extends BaseAction<Area> {
	
	private File file;

	private String fileFileName;

	public void setFileFileName(String fileFileName) {
		this.fileFileName = fileFileName;
	}

	public void setFile(File file) {
		this.file = file;
	}
	
	@Autowired
	private AreaService areaService;
	
	@Action(value="area_pageQuery",results={@Result(name="success",type="json")})
	public String pageQuery(){
		Pageable pageable=new PageRequest(page-1, rows);
		Specification<Area> specification=new Specification<Area>() {

			@Override
			public Predicate toPredicate(Root<Area> root,
					CriteriaQuery<?> query, CriteriaBuilder cb) {
				List<Predicate> list=new ArrayList<>();
				if(StringUtils.isNotBlank(model.getProvince())){
					Predicate p1=cb.like(root.get("province").as(String.class),"%"+ model.getProvince()+"%");
					list.add(p1);
				}
				if(StringUtils.isNotBlank(model.getCity())){
					Predicate p2=cb.like(root.get("city").as(String.class),"%"+ model.getCity()+"%");
					list.add(p2);
				}
				if(StringUtils.isNotBlank(model.getDistrict())){
					Predicate p3=cb.like(root.get("district").as(String.class),"%"+ model.getDistrict()+"%");
					list.add(p3);
				}
				
				return cb.and(list.toArray(new Predicate[0]));
			}			
		};
		Page<Area> pageData=areaService.findPageData(specification,pageable);
		pushPageDataToValueStack(pageData);
		
		return SUCCESS;
	}
	
	
	@Action(value = "area_batchImport")
	public String batchImport() throws IOException {
		String flag = "1";// 标记是否上传成功
		try {
			Workbook workBook = null;

			if (fileFileName.endsWith(".xls")) {// 判断是否是以.xls结尾的文件
				workBook = new HSSFWorkbook(new FileInputStream(file));
				
			} else {// 判断是否是以.xlsx结尾的文件
				workBook = new XSSFWorkbook(new FileInputStream(file));
				
			}
			List<Area> areas = new ArrayList<Area>();// 用于储存读取到的area对象
			Sheet sheet = (Sheet) workBook.getSheetAt(0);
			for (Row row : sheet) {
				if (row.getRowNum() == 0) {// 跳过第一行
					continue;
				}
				if (row.getCell(0) == null
						|| StringUtils.isBlank(row.getCell(0)
								.getStringCellValue())) {
					continue;
				}

				Area area = new Area();
				area.setId(row.getCell(0).getStringCellValue());
				area.setProvince(row.getCell(1).getStringCellValue());
				area.setCity(row.getCell(2).getStringCellValue());
				area.setDistrict(row.getCell(3).getStringCellValue());
				area.setPostcode(row.getCell(4).getStringCellValue());
				String province = area.getProvince();
				String city = area.getCity();
				String district = area.getDistrict();
				city = city.substring(0, city.length() - 1);
				province = province.substring(0, province.length() - 1);
				district = district.substring(0, district.length() - 1);
				String[] headArray = PinYin4jUtils.getHeadByString(province
						+ city + district);
				StringBuilder sb = new StringBuilder();
				for (String string : headArray) {
					sb.append(string);
				}
				String shortcode = sb.toString();
				area.setShortcode(shortcode);
				area.setCitycode(PinYin4jUtils.hanziToPinyin(city, ""));
				areas.add(area);

			}
			areaService.batchImport(areas);
		} catch (Exception e) {
			e.printStackTrace();
			flag = "0";
		}
		ServletActionContext.getResponse().getWriter().write(flag);

		return NONE;
	}
	

}
